package com.igrowing.rpc.springboot.starter.bootstrap;

import com.igrowing.easyRpc.RpcApplication;
import com.igrowing.easyRpc.config.RpcConfig;
import com.igrowing.easyRpc.protocol.ProtocolKeys;
import com.igrowing.easyRpc.server.http.VertxHttpServer;
import com.igrowing.easyRpc.server.tcp.VertxTcpServer;
import com.igrowing.rpc.springboot.starter.annotation.EnableRpc;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.context.annotation.ImportBeanDefinitionRegistrar;
import org.springframework.core.type.AnnotationMetadata;

@Slf4j
public class RpcInitBootstrap implements ImportBeanDefinitionRegistrar {

    /**
     * Spring 初始化时执行，初始化 RPC 框架
     *
     * @param importingClassMetadata
     * @param registry
     * 当一个配置类被@Import注解引用，并且这个配置类实现了ImportBeanDefinitionRegistrar接口时，
     * Spring容器会在启动过程中调用该实现类的registerBeanDefinitions方法。
     * 在这个方法中，可以实现自定义的bean定义注册逻辑，比如根据特定条件动态地注册一些bean
     */
    @Override
    public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
        // 获取 EnableRpc 注解的属性值
        boolean needServer = (boolean) importingClassMetadata.getAnnotationAttributes(EnableRpc.class.getName())
                .get("needServer");

        // RPC 框架初始化（配置和注册中心）
        RpcApplication.init();

        // 全局配置
        final RpcConfig rpcConfig = RpcApplication.getRpcConfig();

        // 启动服务器
        if (needServer) {
            if (ProtocolKeys.TCP_PROTOCOL.equals(rpcConfig.getProtocol())) {
                // 启动TCP服务
                VertxTcpServer vertxTcpServer = new VertxTcpServer();
                vertxTcpServer.doStart(rpcConfig.getServerPort());
            } else {
                // 启动Web服务
                VertxHttpServer httpServer = new VertxHttpServer();
                httpServer.doStart(rpcConfig.getServerPort());
            }


        } else {
            log.info("不启动 server");
        }

    }
}